<template>
  <div class="swagger-container">
    <div class="swagger-header">
      <el-card shadow="never" class="p-4">
        <div class="flex flex-col md:flex-row md:items-center md:justify-between">
          <div>
            <h2 class="text-lg font-bold mb-2">API 文档</h2>
            <p class="text-gray-500">查看和测试系统提供的所有 API 接口</p>
          </div>
          <div v-if="!hasAdminRole" class="mt-2 md:mt-0 text-amber-500">
            <el-icon><Warning /></el-icon>
            <span class="ml-1">您可能需要管理员权限才能访问完整的API文档</span>
          </div>
        </div>
      </el-card>
    </div>

    <div class="swagger-content mt-4">
      <el-card shadow="never">
        <div v-if="loading" class="flex items-center justify-center h-40">
          <el-progress type="circle" />
          <span class="ml-2">加载Swagger文档中...</span>
        </div>
        <iframe
          v-else
          ref="iframeRef"
          class="swagger-iframe"
          @load="onIframeLoad"
          frameborder="0"
        ></iframe>
      </el-card>
    </div>
  </div>
</template>

<script setup lang="ts">
  import { ref, onMounted, onBeforeUnmount } from 'vue'
  import { ElMessage, ElProgress } from 'element-plus'
  import { useUserStore } from '@/store/modules/user'
  import api from '@/utils/http'
  import { Warning } from '@element-plus/icons-vue'

  // 响应式数据
  const userStore = useUserStore()
  const hasAdminRole = ref(true) // 初始值设为true，避免不必要的警告显示
  const loading = ref(true)
  const iframeRef = ref<HTMLIFrameElement>()

  /**
   * 检查用户是否拥有ADMIN角色
   */
  const checkUserRole = async () => {
    try {
      const userInfo = await api.get<{ roles?: Array<{ name: string }> }>({
        url: '/user/info',
        showErrorMessage: false
      })

      // 检查用户是否有ADMIN角色
      hasAdminRole.value = !!userInfo?.roles?.some(
        (role) => role.name === 'ADMIN' || role.name === 'ROLE_ADMIN'
      )

      if (!hasAdminRole.value) {
        ElMessage.warning('您可能需要管理员权限才能访问完整的API文档')
      }
    } catch (error) {
      console.error('获取用户信息失败:', error)
    }
  }

  /**
   * iframe加载完成回调
   */
  const onIframeLoad = () => {
    loading.value = false
    console.log('Swagger文档加载完成')

    // iframe加载完成后，尝试向其传递token
    try {
      if (iframeRef.value?.contentWindow) {
        const token = userStore.accessToken
        if (token) {
          // 向iframe发送消息，传递token
          iframeRef.value.contentWindow.postMessage(
            {
              type: 'SET_AUTH_TOKEN',
              token: token
            },
            '*'
          )
        }
      }
    } catch (error) {
      console.warn('向iframe传递token失败:', error)
    }
  }

  /**
   * 确保token已经设置到请求头和localStorage中
   */
  const ensureTokenIsSet = async () => {
    try {
      const token = userStore.accessToken
      if (!token) {
        ElMessage.warning('请先登录')
        return false
      }

      // 将token存储到localStorage中，这样iframe中的Swagger UI可以读取
      localStorage.setItem('Authorization', token)

      // 发送一个轻量级的请求来确保认证状态，并触发axios拦截器设置请求头
      await api.get({
        url: '/user/info',
        showErrorMessage: false
      })

      console.log('token已设置到请求头和localStorage中')
      return true
    } catch (error) {
      console.error('确保token设置失败:', error)
      ElMessage.error('认证失败，请重新登录')
      return false
    }
  }

  /**
   * 使用request.get请求Swagger页面内容并填充到iframe
   */
  const loadSwaggerContent = async () => {
    try {
      // 使用axios的get请求获取Swagger页面内容
      const response = await api.get<string>({
        url: '../swagger-ui/index.html',
        responseType: 'text',
        showErrorMessage: false
      })

      if (iframeRef.value && response) {
        // 获取iframe的document对象
        const iframeDoc = iframeRef.value.contentDocument || iframeRef.value.contentWindow?.document
        if (iframeDoc) {
          // 填充页面内容
          iframeDoc.open()
          iframeDoc.write(response)
          iframeDoc.close()
          console.log('Swagger页面内容已填充到iframe')
        }
      }
    } catch (error) {
      console.error('获取Swagger页面内容失败:', error)
      ElMessage.error('加载Swagger文档失败')
      loading.value = false
    }
  }

  /**
   * 组件挂载时的初始化逻辑
   */
  onMounted(async () => {
    // 检查用户角色
    await checkUserRole()

    // 确保token已经设置到请求头中
    const isAuthenticated = await ensureTokenIsSet()
    if (!isAuthenticated) {
      loading.value = false
      return
    }

    // 设置加载超时处理
    const timeoutId = setTimeout(() => {
      if (loading.value) {
        loading.value = false
        ElMessage.warning('Swagger文档加载可能超时')
      }
    }, 30000) // 30秒超时

    // 加载Swagger页面内容到iframe
    await loadSwaggerContent()

    // 清理函数
    return () => clearTimeout(timeoutId)
  })

  /**
   * 组件卸载前的清理逻辑
   */
  onBeforeUnmount(() => {
    // 如果iframe存在，可以在这里进行清理
    if (iframeRef.value) {
      iframeRef.value.src = 'about:blank'
    }
  })
</script>

<style lang="scss" scoped>
  .swagger-container {
    display: flex;
    flex-direction: column;
    height: 100vh;
    padding: 16px;

    .swagger-header {
      margin-bottom: 16px;
    }

    .swagger-content {
      display: flex;
      flex: 1;
      flex-direction: column;

      .el-card {
        display: flex;
        flex-direction: column;
        height: 100%;

        .swagger-iframe {
          width: 100%;
          height: 100%;
          min-height: 600px;
        }
      }
    }
  }
</style>
